package com.czk.algorithm.kmp.practice;

/**
 * @Author:ChenZhangKun
 * @Date: 2020/8/7 22:58
 */
@SuppressWarnings("all")
public class Kmp {
    public static void main(String[] args) {
        String str1 = "guishdjshds";
        String str2 = "dj";
        System.out.println(kmp(str1,str2));
    }

    /**
     * kmp算法
     * @param str1
     * @param str2
     * @return
     */
    public static int kmp(String str1,String str2){
        // 拿到str2的部分匹配表
        int[] next = next(str2);
        //
        for (int i = 0,j=0; i < str1.length(); i++) {
            while (j>0&&str1.charAt(i)!=str2.charAt(j)){
                j=next[i-1];
            }
            if (str1.charAt(i)==str2.charAt(j)){
                j++;
            }
            if (j==str2.length()){
                return i-j+1;
            }
        }
        return -1;
    }
    /**
     * 拿到被匹配字符串的部分匹配表
     * @param str 字符串
     * @return 部分匹配表数组
     */
    public static int[] next(String str){
        // c创建数组
        int[] next=new int[str.length()];
        // 第一个为0
        next[0]=0;
        for (int i = 1,j=0; i < str.length(); i++) {
            while (j>0&&str.charAt(i)!=str.charAt(j)){
                j=next[i-1];
            }
            if (str.charAt(i)==str.charAt(j)){
                j++;
            }
          next[i]=j;
        }
        return next;
    }
}
